从零编写一个Koa + graphQL的案例

在Nest.js的文档中看到了有集成GraphQL的指导，所以在本地尝试下先用Koa写出一个DEMO，之后再去与Nest.js集成起来。

先写出数据库模型（这个文件是之前就存在的，没有做更改，将文件名改成了models.ts）：

/**
 * Created by w on 2018/4/13.
 */
const mongoose = require('mongoose');

mongoose.Promise = global.Promise;
mongoose.connect('mongodb://localhost/ticket', {
  server: {
    socketOptions: {
      keepAlive: 1
    }
  }
});

const models = {
  users: {
    username: {
      type: String,
      required: true
    },
    password: {
      type: String,
      required: true
    },
    description: {
      type: String
    },
    createTime: {
      type: Date,
      default: new Date()
    }
  },
  tickets: {
    name: {
      type: String,
      required: true
    },
    price: {
      type: Number,
      requred: true
    },
    holdTime: {   //举办时间
      type: Date,
      required: true
    },
    count: {    //剩余数量
      type: String,
      required: true
    },
    place:{
      type: String,
      required: true
    },
    img: {
      type: String
    },
    description: {
      type: String
    },
    publicTime: {
      type: Date,
      default: new Date()
    },
  }
};

for (let m in models) {
  mongoose.model(m, mongoose.Schema(models[m]));
}

module.exports = {
  getModules: (name) => {
    return mongoose.model(name);
  }
};

之后编写各自模型的GraphQL查询文件（user.ts)）：

// @ts-ignore
const {
  //@ts-ignore
  graphql, GraphQLSchema, GraphQLObjectType, GraphQLString, GraphQLID, GraphQLList, GraphQLNonNull, GraphQLInt, isOutputType,
} = require('graphql');
// @ts-ignore
const User = require('../db/model').getModules('users');

const userType = new GraphQLObjectType({
  name: 'User',
  fields: {
    username: {
      type: GraphQLString,
    },
    password: {
      type: GraphQLString,
    },
    description: {
      type: GraphQLString,
    },
    createTime: {
      type: GraphQLString,
    }
  }
});

module.exports = {
  query: {
    type: new GraphQLList(userType),
    args: {
      username: {
        username: 'username',
        type: GraphQLNonNull(GraphQLString)
      }
    },
    resolve(root, params, options) {
      if (params.username === '$all') {
        return User.find().exec()
      }
      return User.find({ username: params.username }).exec();
    }
  },
  mutate: {
    type: new GraphQLList(userType),
    args: {
      operate: {
        name: 'operate',
        type: GraphQLString
      },
      username: {
        name: 'username',
        type: GraphQLNonNull(GraphQLString)
      },
      possword: {
        name: 'price',
        type: GraphQLNonNull(GraphQLString)
      },
      createTime: {
        name: 'createTime',
        type: GraphQLString,
      },
      description: {
        name: 'description',
        type: GraphQLString
      }
    },
    resolve: async (root, params, options) => {
      try {
        if (params.operate === '$delete') {
          await User.delete({username: params.username});

          return User.find().exec()
        }
        if (params.operate === '$update') {
          await User.update({username: params.username});

          return User.find({ username: params.username }).exec()
        }
        let user = new User({
          username: params.username,
          password: params.password,
          description: params.description,
          createTime: params.createTime
        });
        await user.save();

        return user.find({ username: params.username }).exec()
      } catch (e) {
        console.error("Error while save data: ", e);
      }
    }
  }
}

ticket.ts:

// @ts-ignore
const {
  // @ts-ignore
  graphql, GraphQLSchema, GraphQLObjectType, GraphQLString, GraphQLID, GraphQLList, GraphQLNonNull, GraphQLInt, isOutputType,
} = require('graphql');
// @ts-ignore
const Ticket = require('../db/model').getModules('tickets');

const ticketType = new GraphQLObjectType({
  name: 'Ticket',
  fields: {
    name: {
      type: GraphQLString
    },
    price: {
      type: GraphQLInt
    },
    holdTime: {
      type: GraphQLString
    },
    count: {
      type: GraphQLInt
    },
    place: {
      type: GraphQLString
    },
    img: {
      type: GraphQLString
    },
    description: {
      type: GraphQLString
    },
    publicTime: {
      type: GraphQLString
    }
  }
});

module.exports = {
  query: {
    type: new GraphQLList(ticketType),
    args: {
      name: {
        name: 'name',
        type: GraphQLNonNull(GraphQLString)
      }
    },
    resolve(root, params, options) {
      if (params.name === '$all') {
        return Ticket.find().exec()
      }
      return Ticket.find({ name: params.name }).exec();
    }
  },
  mutate: {
    type: new GraphQLList(ticketType),
    args: {
      operate: {
        name: 'operate',
        type: GraphQLString
      },
      name: {
        name: 'name',
        type: GraphQLNonNull(GraphQLString)
      },
      price: {
        name: 'price',
        type: GraphQLInt
      },
      holdTime: {
        name: 'holdTime',
        type: GraphQLNonNull(GraphQLString)
      },
      count: {
        name: 'count',
        type: GraphQLNonNull(GraphQLInt)
      },
      place: {
        name: 'place',
        type: GraphQLNonNull(GraphQLString)
      },
      img: {
        name: 'img',
        type: GraphQLString
      },
      description: {
        name: 'description',
        type: GraphQLString
      },
      publicTime: {
        name: 'publicTime',
        type: GraphQLString
      }
    },
    resolve: async (root, params, options) => {
      try {
        if (params.operate === '$delete') {
          await Ticket.findOneAndRemove({ name: params.name });

          return Ticket.find().exec();
        }
        if (params.operate === '$update') {
          await Ticket.findByIdAndUpdate({ name: params.name }, {
            name: params.name,
            price: params.price,
            holdTime: params.holdTime,
            count: params.count,
            place: params.place,
            img: params.img,
            description: params.description,
            publicTime: params.publicTime
          });

          return Ticket.find().exec();
        }
        let ticket = new Ticket({
          name: params.name,
          price: params.price,
          holdTime: params.holdTime,
          count: params.count,
          place: params.place,
          img: params.img,
          description: params.description,
          publicTime: params.publicTime
        });
        await ticket.save();

        return Ticket.find({ name: params.name }).exec()
      } catch (e) {
        console.error("Error while save data: ", e);
      }
    }
  }
}

接下来编写用于查询或修改数据的Schema.ts:

const {
  //@ts-ignore
  GraphQLSchema, GraphQLObjectType,
} = require('graphql');
//@ts-ignore
const Ticket = require('./ticket');
//@ts-ignore
const User = require('./user');

module.exports = new GraphQLSchema({
  query: new GraphQLObjectType({
    name: 'Queries',
    fields: {
      Ticket: Ticket.query,
      User: User.query,
    }
  }),
  mutation: new GraphQLObjectType({
    name: 'Mutations',
    fields: {
      Ticket: Ticket.mutate,
      User: User.mutate,
    }
  })
})

编写服务端(server.ts)：

const Koa = require('koa');
const app = new Koa();
const koaBody = require('koa-body');
const CombileRouter = require('./combineRouter');

app
  .use(koaBody({
    multipart: true,
    formidable: {
      keepExtensions: true,
    },
  }))
  .use(CombileRouter.routes())
  .use(CombileRouter.routes());

app.listen(5000, () => {
  console.log('App running~');
})

定义ticket路由的ticket.router.ts:

const KoaRouter = require('koa-router');
const ticketRouter = new KoaRouter();
// @ts-ignore
const { graphqlKoa, graphiqlKoa } = require('graphql-server-koa');
// @ts-ignore
const Ticket = require('../db/model').getModules('tickets');
const ticketSchema = require('../graphql/schema');

ticketRouter.get('/ticket/all', async (ctx) => {
  try {
    let result = await Ticket.find({});

    ctx.body = { data: Object.assign({}, result), code: '0' };
  } catch (e) {
    console.error("Getting all ticket error: ", e);
  }
});

ticketRouter.post('/graphql', async (ctx, next) => {
  console.log('graphql', ctx.request.body);
  await graphqlKoa({schema: ticketSchema})(ctx, next)
})
.get('/graphql', async (ctx, next) => {
  await graphqlKoa({schema: ticketSchema})(ctx, next)
})
.get('/graphiql', async (ctx, next) => {
  await graphiqlKoa({endpointURL: '/graphql'})(ctx, next)
})


module.exports = ticketRouter;

还有之前定义的user.router.ts(测试这个user路由是否是reachable的):

const Router = require('koa-router');
const userRouter = new Router();
// @ts-ignore
const { graphiqlKoa } = require('graphql-server-koa');

userRouter.get('/user/all', async (ctx) => {
  ctx.body = {
    code: '0',
    msg: 'OK',
    info: {
      data: [{
        username: 'a',
        id: 0,
        desc: 'ok',
      }, {
        username: 'd',
        id: 1,
        desc: 'ok',
      }, {
        username: 'c',
        id: 2,
        desc: 'ok',
      }, {
        username: 'b',
        id: 3,
        desc: 'ok',
      }]
    }
  }
});

module.exports = userRouter;

将两个路由文件融合起来(combineRouter.ts):

const UserRouter:object = require('./koa-router/user.router');
const TicketRouter:object = require('./koa-router/ticket.router');

const combiler = (routers: object[]): object => {
  let arr = [];
  
  routers.forEach(v => {
    //@ts-ignore
    arr.push(...v.stack)
  });
  return Object.assign(routers[0], { stack:arr });
}

const res = combiler([UserRouter, TicketRouter]);

module.exports = res;

OK了，这部分就写好了基础的query和mutation功能，还有删除的功能之后再加上。
附上tsconfig.json文件：

{
  "compilerOptions": {
    "module": "commonjs",
    "target": "es6",
    "sourceMap": true,
    "experimentalDecorators": true,
  },
  "exclude": [
    "node_modules"
  ]
}